How to check what a function prints and do something off of that